package 二叉树;

import java.util.*;

/**
 * @author liucong
 * @date 2021/3/1 - 17:57
 *
 * 给定一个二叉树，返回其节点值自底向上的层序遍历。 （即按从叶子节点所在层到根节点所在的层，逐层从左向右遍历）
 *
 * 例如：
 * 给定二叉树 [3,9,20,null,null,15,7],
 *
 *     3
 *    / \
 *   9  20
 *     /  \
 *    15   7
 * 返回其自底向上的层序遍历为：
 *
 * [
 *   [15,7],
 *   [9,20],
 *   [3]
 * ]
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class _107_二叉树的层序遍历_II {
    //链表头部添加
    public List<List<Integer>> levelOrderBottom2(TreeNode root) {
        List<List<Integer>> list = new ArrayList<>();

        if(root == null) return list;

        Queue<TreeNode> queue = new LinkedList<>();
        List<Integer> list1 = new ArrayList<>();
        queue.offer(root);
        int levelSize = 1;
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();
            list1.add(node.val);
            levelSize--;

            if(node.left != null){
                queue.offer(node.left);
            }
            if(node.right != null){
                queue.offer(node.right);
            }
            if(levelSize == 0){
                list.add(0, list1);
                list1 = new ArrayList<>();
                levelSize = queue.size();
            }
        }
        return list;
    }

    //使用栈进行反转
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        List<List<Integer>> list = new ArrayList<>();

        if(root == null) return list;

        Queue<TreeNode> queue = new LinkedList<>();
        Stack<List<Integer>> stack = new Stack<>();
        List<Integer> list1 = new ArrayList<>();
        queue.offer(root);
        int levelSize = 1;
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();
            list1.add(node.val);
            levelSize--;

            if(node.left != null){
                queue.offer(node.left);
            }
            if(node.right != null){
                queue.offer(node.right);
            }
            if(levelSize == 0){
                stack.push(list1);
                list1 = new ArrayList<>();
                levelSize = queue.size();
            }
        }
        while(!stack.isEmpty()){
            list.add(stack.pop());
        }
        return list;
    }
}
